import os

# 获取数据集视频的类别数，并根据类名创建相应的文件夹
cls_name = {'UAV', 'OTHER'}
target_dir = r"E:\uav_datasets\Drone_detection_dataset-UAV2\\"
# 获取分类名，组成集合
print(cls_name)

# 创建每个分类的目录
for cls_nm in cls_name:
    os.mkdir(target_dir + cls_nm)
